package 搜索.广度优先搜索.easy;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

/**
 * @author Lw中
 * @date 2020/8/6 11:35
 */
public class JzOffer32 {


     public class TreeNode {
          int val;
          TreeNode left;
          TreeNode right;
          TreeNode(int x) { val = x; }
      }

    /**
     * 使用广度优先搜索的方法完成树的遍历
      * @param root
     * @return
     */
    public List<List<Integer>> levelOrder(TreeNode root) {
        // 二维数组
        List<List<Integer>> lists = new ArrayList<List<Integer>>();
        // 新建一个用链实现的队列用于存储节点
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        // 判断是否为空二叉树
        if(root != null) {
            queue.add(root);
        }
        while (!queue.isEmpty()) {
            //创建arrayList用于存储每一层节点
            List<Integer> list = new ArrayList<>();
            for (int i = queue.size(); i > 0 ; i--) {
                TreeNode node = queue.poll();
                list.add(node.val);
                // 如果左节点不为空入列
                if (node.left != null) {
                    queue.offer(node.left);
                }
                // 如果右节点不为空入列
                if (node.right != null) {
                    queue.offer(node.right);
                }
            }
            lists.add(list);
        }
        return lists;
    }
}
